#pragma once

#include <iostream>
#include <string>
#include <queue>
#include <vector>
#include <memory>
#include "Log.hpp"
#include "Mutex.hpp"
#include "Cond.hpp"
#include "Thread.hpp"

namespace ThreadPoolModule
{
    using namespace LogModule;
    using namespace ThreadModule;
    using namespace LockModule;
    using namespace CondModule;

    // 用来做测试的线程方法
    void Jyz()
    {
        while (true)
        {
            LOG(LogLevel::DEBUG) << "我喜欢王乐乐";
            sleep(1);
        }
    }

    using thread_t = std::shared_ptr<Thread>;

    const static int defaultnum = 5;

    template <typename T>
    class ThreadPool
    {
    private:
        // 判断队列是否为空
        bool IsEmpty() { return _taskq.empty(); }

        // 线程要执行的任务
        void HandlerTask(std::string name)
        {
            LOG(LogLevel::INFO) << "线程: " << name << ", 进入HandlerTask的逻辑";
            while (true)
            {
                // 1. 拿任务
                T t;
                {
                    LockGuard lockguard(_lock);
                    while (IsEmpty() && _isrunning)
                    {
                        _wait_num++;
                        _cond.Wait(_lock);
                        _wait_num--;
                    }
                    // 2. 任务队列为空 && 线程池退出了
                    if (IsEmpty() && !_isrunning)
                        break;

                    t = _taskq.front();
                    _taskq.pop();
                }
                // 2. 处理任务
                t(); // 规定，未来所有的任务处理，全部都是必须提供()方法！
            }
            LOG(LogLevel::INFO) << "线程: " << name << " 退出";
        }

        ThreadPool(const ThreadPool<T> &) = delete;
        ThreadPool<T> &operator=(const ThreadPool<T> &) = delete;

        ThreadPool(int num = defaultnum)
            : _num(num), _wait_num(0), _isrunning(false)
        {
            // 创建线程（不启动）
            for (int i = 0; i < _num; i++)
            {
                //_threads.push_back(std::make_shared<Thread>(Jyz));

                _threads.push_back(std::make_shared<Thread>(std::bind(&ThreadPool::HandlerTask, this, std::placeholders::_1)));
                LOG(LogLevel::INFO) << "构建线程" << _threads.back()->Name() << "对象... 成功";
            }
        }

    public:
        ~ThreadPool() {}

        // 单例的创建方式
        static ThreadPool<T> *getInstance()
        {
            if (instance == NULL)
            {
                LockGuard lockguard(mutex);
                if (instance == NULL)
                {
                    LOG(LogLevel::INFO) << "单例首次被执行，需要加载对象...";
                    instance = new ThreadPool<T>();
                    //创建出来以后直接启动
                    instance->Start();
                }
            }
            return instance;
        }

        // 入队列
        void Equeue(T &in)
        {
            LockGuard lockguard(_lock);
            if (!_isrunning)
                return;
            _taskq.push(in);
            if (_wait_num > 0)
            {
                // 唤醒线程
                _cond.Notify();
            }
        }

        // 启动
        void Start()
        {
            if (_isrunning)
                return;
            _isrunning = true;
            for (auto &n : _threads)
            {
                n->Start();
                LOG(LogLevel::INFO) << "启动线程" << n->Name() << "... 成功";
            }
            _isrunning = true;
        }

        // 等待
        void Wait()
        {
            for (auto &n : _threads)
            {
                n->Join();
                LOG(LogLevel::INFO) << "回收线程" << n->Name() << "... 成功";
            }
        }
        // 停止
        void Stop()
        {
            LockGuard lockguard(_lock);
            if (_isrunning)
            {
                // 3.任务队列不能再入队列了
                _isrunning = false; // 不工作状态
                // 1.让线程自己退出（要唤醒）&&2.任务队列的历史任务被处理完了
                if (_wait_num > 0)
                {
                    _cond.NotifyAll();
                }
            }
        }

    private:
        std::vector<thread_t> _threads; // 线程数组
        int _num;                       // 线程数量
        std::queue<T> _taskq;           // 任务队列
        Mutex _lock;                    // 锁
        Cond _cond;                     // 互斥量
        int _wait_num;                  // 等待的线程数量
        bool _isrunning;                // 线程池是否退  flase->退出，true->运行

        // 类内定义
        static ThreadPool<T> *instance;
        static Mutex mutex; // 只用来保护单例
    };

    // 类外声明
    template <typename T>
    ThreadPool<T> *ThreadPool<T>::instance = NULL;
    template <typename T>
    Mutex ThreadPool<T>::mutex; // 只用来保护单例

}